// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef MEDIA_BLINK_TEST_RESPONSE_GENERATOR_H_
#define MEDIA_BLINK_TEST_RESPONSE_GENERATOR_H_

#include <stdint.h>

#include "base/macros.h"
#include "third_party/WebKit/public/platform/WebURLError.h"
#include "third_party/WebKit/public/platform/WebURLResponse.h"
#include "url/gurl.h"

namespace media {

// Generates WebURLErrors and WebURLResponses suitable for testing purposes.
class TestResponseGenerator {
public:
    enum Flags {
        kNormal = 0,
        kNoAcceptRanges = 1 << 0, // Don't include Accept-Ranges in 206 response.
        kNoContentRange = 1 << 1, // Don't include Content-Range in 206 response.
        kNoContentLength = 1 << 2, // Don't include Content-Length in 206 response.
        kNoContentRangeInstanceSize = 1 << 3, // Content-Range: N-M/* in 206.
    };

    // Build an HTTP response generator for the given URL. |content_length| is
    // used to generate Content-Length and Content-Range headers.
    TestResponseGenerator(const GURL& gurl, int64_t content_length);

    // Generates a WebURLError object.
    blink::WebURLError GenerateError();

    // Generates a regular HTTP 200 response.
    blink::WebURLResponse Generate200();

    // Generates a regular HTTP 206 response starting from |first_byte_offset|
    // until the end of the resource.
    blink::WebURLResponse Generate206(int64_t first_byte_offset);

    // Generates a custom HTTP 206 response starting from |first_byte_offset|
    // until the end of the resource. You can tweak what gets included in the
    // headers via |flags|.
    blink::WebURLResponse Generate206(int64_t first_byte_offset, Flags flags);

    // Generates a regular HTTP 206 response starting from |first_byte_offset|
    // until |last_byte_offset|.
    blink::WebURLResponse GeneratePartial206(int64_t first_byte_offset,
        int64_t last_byte_offset);

    // Generates a custom HTTP 206 response starting from |first_byte_offset|
    // until |last_byte_offset|. You can tweak what gets included in the
    // headers via |flags|.
    blink::WebURLResponse GeneratePartial206(int64_t first_byte_offset,
        int64_t last_byte_offset,
        Flags flags);

    // Generates a regular HTTP 404 response.
    blink::WebURLResponse Generate404();

    // Generate a HTTP response with specified code.
    blink::WebURLResponse GenerateResponse(int code);

    // Generates a file:// response starting from |first_byte_offset| until the
    // end of the resource.
    //
    // If |first_byte_offset| is negative a response containing no content length
    // will be returned.
    blink::WebURLResponse GenerateFileResponse(int64_t first_byte_offset);

    int64_t content_length() { return content_length_; }

private:
    GURL gurl_;
    int64_t content_length_;

    DISALLOW_COPY_AND_ASSIGN(TestResponseGenerator);
};

} // namespace media

#endif // MEDIA_BLINK_TEST_RESPONSE_GENERATOR_H_
